#!/usr/bin/env python  
# encoding: utf-8
import argparse

parser = argparse.ArgumentParser(description='PyTorch Airway Segmentation')
parser.add_argument('--model', '-m', metavar='MODEL', default='baseline_fr',
					help='model')
parser.add_argument('-j', '--workers', default=4, type=int, metavar='N',
					help='number of data loading workers (default: 4)')
parser.add_argument('-b', '--batch-size', default=2, type=int,
					metavar='N', help='mini-batch size (default: 2)')
parser.add_argument('--inputpath', default=None, type=str, metavar='input',
					help='input directory path')
parser.add_argument('--outputpath', default=None, type=str, metavar='output',
					help='directory to save prediction results')
parser.add_argument('--debugdataloader', default=0, type=int, metavar='DebugDataloader',
					help='debug mode')
parser.add_argument('--cubesize', default=[80,192,304], nargs="*", type=int, metavar='cube',
					help='cube size')
parser.add_argument('--stridev', default=[32,32,32], nargs="*", type=int, metavar='stride',
					help='split stride val')
parser.add_argument('--multigpu', default=False, type=bool, metavar='mgpu',
					help='use multiple gpus')
parser.add_argument('--preprocess', default=0, type=int, metavar='PreprocessImage',
					help='use preprocessing')
parser.add_argument('--threshold', '-th', default=0.5, type=float, metavar='THRESHOLD',
					help='use threshold to binarize prediction results')
